"""
动态GDP柱状图演示
"""

from pyecharts.charts import Bar, Timeline
from pyecharts.options import *
from pyecharts.globals import ThemeType
import os

# 打开文件，读取数据，关闭文件
f = open('D:\Document\python学习\资料\可视化案例数据\动态柱状图数据\\1960-2019全球GDP数据.csv', 'r', encoding='GB2312')
data_lines = f.readlines()
f.close()

# 去掉第一行数据
data_lines.pop(0)

# 创建字典，处理数据
data_dict = {}
for line in data_lines:
    year = int(line.split(',')[0])
    country = line.split(',')[1]
    gdp = float(line.split(',')[2])
    try:
        data_dict[year].append([country, gdp])
    except KeyError:
        data_dict[year] = []
        data_dict[year].append([country, gdp])

# 创建时间线对象
timeline = Timeline({"theme": ThemeType.LIGHT})

# 排序年份
sorted_year_list = sorted(data_dict.keys())

# 循环获取每年 gdp 最高的8个国家
for year in sorted_year_list:
    data_dict[year].sort(key=lambda e: e[1], reverse=True)
    year_data = data_dict[year][:8]
    x_data = []
    y_data = []
    for country_gdp in year_data:
        x_data.append(country_gdp[0])
        y_data.append(country_gdp[1] / 100000000)

    # 构建柱状图
    bar = Bar()
    x_data.reverse()
    y_data.reverse()
    bar.add_xaxis(x_data)
    bar.add_yaxis('GDP(亿)', y_data, label_opts=LabelOpts(position='right'))
    bar.reversal_axis()
    bar.set_global_opts(
        title_opts=TitleOpts(title=f'{year}年全球GDP前8数据')
    )
    timeline.add(bar, str(year))

# 设置自动播放
timeline.add_schema(
    play_interval=500,
    is_timeline_show=True,
    is_auto_play=True,
    is_loop_play=True
)

# 生成图表
timeline.render('1960-2019全球GDP前8数据.html')

os.system('1960-2019全球GDP前8数据.html')
